package com.dreamoo.common.enums;

import lombok.Getter;

/**
 * 模型类型枚举
 *
 * @author dreamoo
 */
@Getter
public enum ModelTypeEnum {
    
    /**
     * 绘画模型
     */
    TEXT("text", "绘画模型"),
    
    /**
     * 解梦模型
     */
    INTERPRETATION("interpretation", "解梦模型"),
    
    /**
     * 续写模型
     */
    CONTINUATION("continuation", "续写模型");
    
    /**
     * 码值
     */
    private final String code;
    
    /**
     * 描述
     */
    private final String desc;
    
    ModelTypeEnum(String code, String desc) {
        this.code = code;
        this.desc = desc;
    }
    
    /**
     * 根据码值获取枚举
     *
     * @param code 码值
     * @return 枚举
     */
    public static ModelTypeEnum getByCode(String code) {
        if (code == null) {
            return TEXT;
        }
        for (ModelTypeEnum value : values()) {
            if (value.getCode().equals(code)) {
                return value;
            }
        }
        return TEXT;
    }
} 